from .optimizer import CustomFp16OptimizerHook